#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <assert.h>
#include <unistd.h>
#include <errno.h>
#include <string.h>
#include <stdio.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <fcntl.h>
#include <stdlib.h>
#include<signal.h>

void handle_signal(int sig)
{
    printf("client recv a signal %d\n", sig);
    exit(1);
}
int main()
{
    int sockfd = socket(PF_INET, SOCK_STREAM, 0);
    if (sockfd < 0)
    {
        perror("socket:");
        exit(1);
    }

    struct sockaddr_in servaddr;
    memset(&servaddr, 0, sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_addr.s_addr = inet_addr("127.0.0.1");
    servaddr.sin_port = htons(5188);

    if ((connect(sockfd, (struct sockaddr *)&servaddr, sizeof(servaddr))) < 0)
    {
        perror("connect:");
        exit(1);
    }
    pid_t pid = fork();
    if (pid < 0)
    {
        perror("fork:");
        exit(1);
    }
    else if (pid == 0)
    {
        signal(SIGUSR1,handle_signal); //客户端注册一个信号，当服务端关闭时候，能够通知到该子进程退出
        char sendbuf[1024] = {0}; //用来发送到网络sockfd的

        //客户端接收键盘数据放到sendbuf
        while (fgets(sendbuf, sizeof(sendbuf), stdin) != NULL)
        {
            write(sockfd, sendbuf, sizeof(sendbuf)); //客户端的数据发送给服务端
            memset(sendbuf, 0, sizeof(sendbuf));
        }
        close(sockfd);
    }
    else
    {
        char recvbuf[1024];
        while (1)
        {
            memset(&recvbuf, 0, sizeof(recvbuf));
            int sz = read(sockfd, recvbuf, sizeof(recvbuf)); //客户端接收服务器发来的数据
            if (sz < 0)
            {
                perror("read");
                exit(1);
            }
            else if (sz == 0)
            {
                printf("peer close\n");
                break;
            }
            fputs(recvbuf, stdout); //客户端打印服务器发来的数据
        }
        close(sockfd);
        kill(pid,SIGUSR1);
        exit(1);
    }

    return 0;
}